package com.zxtc.wh.utils;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.util.List;
import java.util.Map;

/**
 * 文件上传
 * 2017年4月13日14:19:09
 */
public class FileUploadSaveUtil {
    private static String tempFileDir = System.getProperty("java.io.tmpdir");

    // 上传文件存储目录
    private static final String UPLOAD_DIRECTORY = "upload";

    // 上传配置
    private static final int MEMORY_THRESHOLD = 1024 * 1024 * 3;  // 3MB
    private static final int MAX_FILE_SIZE = 1024 * 1024 * 40 * 10; // 40MB*10
    private static final int MAX_REQUEST_SIZE = 1024 * 1024 * 50 * 10; // 50MB*10


    public static List<FileItem> getFileParameter(HttpServletRequest request) throws FileUploadException {
        return getDefaultServletFileUpload().parseRequest(request);
    }

    public static ServletFileUpload getDefaultServletFileUpload() {
        // 配置上传参数
        DiskFileItemFactory factory = new DiskFileItemFactory();
        // 设置内存临界值 - 超过后将产生临时文件并存储于临时目录中
        factory.setSizeThreshold(MEMORY_THRESHOLD);
        // 设置临时存储目录
        factory.setRepository(new File(tempFileDir));

        ServletFileUpload upload = new ServletFileUpload(factory);

        // 设置最大文件上传值
        upload.setFileSizeMax(MAX_FILE_SIZE);

        // 设置最大请求值 (包含文件和表单数据)
        upload.setSizeMax(MAX_REQUEST_SIZE);
        return upload;
    }

    public static Map<String, List<FileItem>> getFileParameterMap(HttpServletRequest request) throws FileUploadException {
        return getDefaultServletFileUpload().parseParameterMap(request);
    }


    /**
     * 保存文件到服务器
     *
     * @param outFile  输出文件
     * @param fileItem 要
     */
    public static void saveFile(File outFile, FileItem fileItem) throws Exception {
        // 保存文件到硬盘
        fileItem.write(outFile);
    }

    /**
     * 在服务端的文件上传保存目录下保存为随机文件名的文件
     *
     * @return 保存的资源文件的相对索引，可以用{@link #getResourceFile(ServletContext, String)} 得到资源文件
     */
    public static String saveRandomName(ServletContext context, FileItem fileItem) throws Exception {
        // 构造临时路径来存储上传的文件

        // 这个路径相对当前应用的目录
        String uploadPath = getResourceDirectory(context);
        File dir = new File(uploadPath);
        dir.mkdirs();
        String suffix = fileItem.getName();
        String prefix = "temp";
        File temp = File.createTempFile(prefix, suffix, dir);
        saveFile(temp, fileItem);

        System.out.println(temp.getCanonicalPath());
        System.out.println(temp.getPath());
        System.out.println(temp.getParent());
        System.out.println(temp.getName());
        System.out.println(FileUploadSaveUtil.class.getClassLoader().getResource("/").getPath() + "/" + temp.getName());

        return temp.getName();
    }

    /**
     * 返回文件的不包含前缀的相对路径,得到
     *
     * @param name 不含任何目录的结构文件名
     * @return
     */
    public static String getFileName(String name) {
        return UPLOAD_DIRECTORY + File.separator + name;

    }


    /**
     * 获取资源文件，功能与{@link #getResourceAbsolutePath(ServletContext, String)} 相似
     *
     * @param servletConfig web容器的配置
     * @param relativePath  资源文件的相对索引
     * @return
     */
    public static File getResourceFile(ServletContext servletConfig, String relativePath) {
        return new File(getResourceAbsolutePath(servletConfig, relativePath));
    }


    /**
     * 获取资源文件的绝对路径
     *
     * @param context      web容器的配置
     * @param relativePath 资源文件的相对索引
     * @return
     */
    public static String getResourceAbsolutePath(ServletContext context, String relativePath) {
        return getResourceDirectory(context) + File.separator + relativePath;
    }


    /**
     * 获取保存资源文件的目录
     *
     * @param context web容器的配置
     */
    public static String getResourceDirectory(ServletContext context) {
        String realPath;

        realPath = context.getContextPath();
        realPath = context.getRealPath("/");
        return realPath + UPLOAD_DIRECTORY;
    }

    /**
     * 获取字段名称，例如：<input type=”text” name=”username”/>，返回的是username
     */
    public static String getFieldName(FileItem fileItem) {
        return fileItem.getFieldName();
    }


}
